import torchvision
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("log1")
dataset_transform = torchvision.transforms.Compose([
    torchvision.transforms.ToTensor()
])
train_set = torchvision.datasets.CIFAR10(root="./datavision",train=True,transform=dataset_transform,download=True)
test_set = torchvision.datasets.CIFAR10(root="./datavision",train=False,transform=dataset_transform,download=True)

print(test_set[0])
writer.add_image("CIFAR10",test_set[0][0])
writer.close()

